log function in python using numpy